# 在本例中我们画出阶跃函数的图像
# 在本例中一定要使用数组元素作为参数来参与运算
import matplotlib.pyplot as plt
import numpy as np


def step_function(x):
    y = x > 0  # 这里一定要将y同时也设置为narray数组才能使用numpy中的相关方法
    y = y.astype(np.int)  # 将逻辑运算后的结果转换为int类型的数据，同时输出，巧妙的实现阶跃函数
    return y


if __name__ == '__main__':
    X = np.arange(-5, 5, 0.1)
    Y = step_function(X)

    plt.plot(X, Y, label='step_function')
    plt.xlabel('X')
    plt.ylabel('Y')

    plt.title('step_function')
    plt.legend()

    plt.show()
